from codex import *

choice = 0

my_list = [
    PINK,
    GREEN,
    "Ahoy",
    pics.HEART, 
    pics.HEART_SMALL, 
    pics.HAPPY
]

# Define the last index
LAST_INDEX = len(my_list) - 1

while True:
    my_image = my_list[choice]

    # If the type is a color
    if type(my_image) == tuple:
        display.fill(my_image)
    else:
        display.show(my_image)

    if buttons.was_pressed(BTN_A):
        choice = choice + 1
        if choice > LAST_INDEX:
            choice = 0

    if buttons.was_pressed(BTN_R):
        choice = choice - 1
        if choice < 0:
            choice = LAST_INDEX

    if buttons.was_pressed(BTN_B):
        choice = choice + 2
        if choice > LAST_INDEX:
            choice = 0

    if buttons.was_pressed(BTN_L):
        choice = choice - 2
        if choice < 0:
            choice = LAST_INDEX

    if buttons.was_pressed(BTN_U):
        choice = choice + 3
        if choice > LAST_INDEX:
            choice = 0

    if buttons.was_pressed(BTN_D):
        choice = choice - 2
        if choice < 0:
            choice = LAST_INDEX